how to convert a numpy array to pil image applying matplotlib colormap - python